#PYOPENGL_PLATFORM=egl python generate_mask_dataset.py --save_tensors
#python generate_mask_dataset.py --save_tensors
python generate_dataset.py --save_tensors
